"""
要素召回测试
"""
from core.milvus_vector import MilvusVector
from loguru import logger


if __name__ == '__main__':

    milvus = MilvusVector()

    logger.info("开始混合检索.")
    documents = milvus.hybrid_search(
        collection_name='dad6204ebdb84513b3afb175e462b', query='争议解决', top_k=10
    )
    logger.info("结束混合检索.")

    for doc in documents:
        print(doc)
